使用YOLOv3训练自己的数据集


声明:本文使用的YOLOv3来自github中的AlexeyAB大神的代码。
系统环境:ubuntu18.04,GTX1660ti
github上有完整的实现步骤,博主只是记录一下。



1.准备自己的数据集

准备好自定义的训练集/验证集/测试集,本文按照PASCAL VOC(COCO也可以)数据集格式进行存储数据,VOC格式如何存储百度一下即可。参考:制作自己的数据集

2.下载及修改代码

2.1下载代码并编译

git clone https://github.com/AlexeyAB/darknet

打开终端运行上面代码,下载到本地(如果网络不好下载太慢,可以上传到码云然后git clone 码云链接)

cd darknet

切换到darknet(目录名)目录中,输入make进行编译
注意:在make之前,找到makefile文件,我们可以设置一些功能,根据自己的需求设置,GPU/CUDNN/OPENCV是必要的
在这里插入图片描述编译完成后运行./build.sh,用于查找已安装的可依赖项,例如:CUDA、cudnn、zed sdk等。(提提前安装好cmake)

2.2根据自己的数据集修改cfg文件

在这里插入图片描述
超参数可以不用修改,分辨率可以提高一点,首次训练可以将max_batches调小一点,等跑通之后再调高。这里的subi

  • 3
    点赞
  • 51
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值